from MODEL_CKP import FLUX_REDUX,DEPTH_PREDCITION
from image_gen_aux import DepthPreprocessor

def load_depth_processor():
    depth_processor = DepthPreprocessor.from_pretrained(DEPTH_PREDCITION)
    return depth_processor

def get_depth_by_processor(depth_processor,img_pil):
    control_image = depth_processor(
            img_pil
        )[0].convert("RGB") # PIL 768 1024
    return control_image